import torch.nn as nn

class LossFunctions:
    def classification_loss(self):
        return nn.CrossEntropyLoss()

    def regression_loss(self):
        return nn.MSELoss()